from loadDatas import loadDatas
from sklearn.tree import DecisionTreeClassifier
from sklearn.ensemble import AdaBoostClassifier


X_train, X_test, y_train, y_test = loadDatas('wine', test_size=0.3)
num_models = 10

model = AdaBoostClassifier(DecisionTreeClassifier(), n_estimators=num_models)
model.fit(X_train, y_train)
trainScore = model.score(X_train, y_train)
testScore = model.score(X_test, y_test)
print(f'\ttrainScore:{trainScore}\n\ttestScore:{testScore}')